package TabelModels;

import java.util.ArrayList;

import javax.swing.table.AbstractTableModel;

import model.Dier;
import Lijsten.ReptielLijst;
import Lijsten.VogelLijst;
import Lijsten.ZoogDierLijst;
import datalayer.*;



@SuppressWarnings("serial")
public class DierentuinTabelModel extends AbstractTableModel{
	private String[] names = {"Soort"};
	private ArrayList<Dier> dieren = new ArrayList<>();
	
	
	
	
	public DierentuinTabelModel(String soort) {
		if(soort.equals("Zoogdier")){
			
			dieren = ZoogDierLijst.getInstance().getDier();
		}
		else if(soort.equals("Reptiel")){
			
			dieren = ReptielLijst.getInstance().getDier();
		}
		else if(soort.equals("Vogel")){
			dieren = VogelLijst.getInstance().getDier();
		}
		else{
			
		}
		
	}
	

	
	public DierentuinTabelModel() {
		// TODO Auto-generated constructor stub
	}



	public int getColumnCount() {
		
		return 1;
	}

	public int getRowCount() {
		
		return dieren.size();
	}

	
	public String getColumnName(int arg0) {
		
		return names[arg0];
	}

	public Object getValueAt(int arg0, int arg1) {
		return dieren.get(arg0).getClass().getSimpleName();
	}
}
